package response

import (
	"io"

	"github.com/kataras/iris/v12"
)

// 发送文件流到客户端
// func SendStream(ctx iris.Context, writer func(w io.Writer) error) {
// 	ctx.Header("Content-Disposition", "attachment; filename=export.xlsx")
// 	ctx.ContentType("application/vnd.openxmlformats-officedocument.spreadsheetml.sheet")
// 	ctx.StreamWriter(writer)
// }

// 发送Excel文件流
//
//	@param src 				文件名称
//	@param writer 			文件流
func SendExcelStream(ctx iris.Context, src string, writer func(w io.Writer) error) {
	ctx.Header("Content-Disposition", "")
	ctx.Header("Content-Disposition", "attachment; filename="+src)

	ctx.ContentType("application/vnd.openxmlformats-officedocument.spreadsheetml.sheet")

	writer(ctx.ResponseWriter())
}

// 发送文件到客户端
//
//	@param 	src				文件地址
//	@param	destName		下载文件名
func SendFile(ctx iris.Context, src string, destName string) {
	ctx.SendFile(src, destName)
}
